stack in python - geeksforgeeks